Backdrop

백준 온라인 저지 ▸ 21736

헌내기는 친구가 필요해
II

문제

2020년에 입학한 헌내기 도연이가 있다. 도연이는 비대면 수업 때문에 학교에 가지 못해 학교에 아는 친구가 없었다. 드디어 대면 수업을 하게 된 도연이는 어서 캠퍼스 내의 사람들과 친해지고 싶다.

도연이가 다니는 대학의 캠퍼스는 N×MN \times M 크기이며 캠퍼스에서 이동하는 방법은 벽이 아닌 상하좌우로 이동하는 것이다. 예를 들어, 도연이가 (xx, yy)에 있다면 이동할 수 있는 곳은 (x+1x+1, yy), (xx, y+1y+1), (x1x-1, yy), (xx, y1y-1)이다. 단, 캠퍼스의 밖으로 이동할 수는 없다.

불쌍한 도연이를 위하여 캠퍼스에서 도연이가 만날 수 있는 사람의 수를 출력하는 프로그램을 작성해보자.

입력

첫째 줄에는 캠퍼스의 크기를 나타내는 두 정수 NN (1N600 1 \leq N \leq 600), MM (1M600 1 \leq M \leq 600)이 주어진다.

둘째 줄부터 NN개의 줄에는 캠퍼스의 정보들이 주어진다. O는 빈 공간, X는 벽, I는 도연이, P는 사람이다. I가 한 번만 주어짐이 보장된다.

출력

첫째 줄에 도연이가 만날 수 있는 사람의 수를 출력한다. 단, 아무도 만나지 못한 경우 TT를 출력한다.

예제 입력 1

3 5
OOOPO
OIOOX
OOOXP

예제 출력 1

1

예제 입력 2

3 3
IOX
OXP
XPP

예제 출력 2

TT

풀이

코드

from collections import deque
 
 
n, m = map(int, input().split())
 
q = deque()
campus = []
for i in range(n):
    row = list(input())
 
    if "I" in row:
        j = row.index("I")
        q.append((i, j))
        row[j] = "X"
 
    campus.append(row)
 
directions = [(1, 0), (0, 1), (-1, 0), (0, -1)]
count = 0
while q:
    x, y = q.popleft()
    for dx, dy in directions:
        a = x + dx
        b = y + dy
        if 0 <= a < n and 0 <= b < m and campus[a][b] != "X":
            if campus[a][b] == "P":
                count += 1
 
            q.append((a, b))
            campus[a][b] = "X"
 
print(count or "TT")